diff --git a/timesketch/lib/analyzers/feature_extraction_test.py b/timesketch/lib/analyzers/feature_extraction_test.py
index 66e66e6238..6ce8529e8e 100644
--- a/timesketch/lib/analyzers/feature_extraction_test.py
+++ b/timesketch/lib/analyzers/feature_extraction_test.py
@@ -22,7 +22,7 @@
 import mock
 
 from timesketch.lib import emojis
-from timesketch.lib.analyzers.feature_extraction_plugin import (
+from timesketch.lib.analyzers.feature_extraction import (
     FeatureExtractionSketchPlugin,
 )
 from timesketch.lib.analyzers.feature_extraction_plugins import regex_features