diff --git a/data/schemaFuncs.py b/data/schemaFuncs.py
index 2e8f9b29..9eb4d091 100644
--- a/data/schemaFuncs.py
+++ b/data/schemaFuncs.py
@@ -1,9 +1,9 @@
 import pywt
 import six
 
-from radiomics.featureextractor import RadiomicsFeaturesExtractor
+from radiomics import getFeatureClasses
 
-featureClasses = RadiomicsFeaturesExtractor.getFeatureClasses()
+featureClasses = getFeatureClasses()
 
 def checkWavelet(value, rule_obj, path):
   if not isinstance(value, six.string_types):