diff --git a/javascript/ql/experimental/adaptivethreatmodeling/test/generic_feature_testing/TestUtil.qll b/javascript/ql/experimental/adaptivethreatmodeling/test/generic_feature_testing/TestUtil.qll
index 46c5877390b..3e035b91ec9 100644
--- a/javascript/ql/experimental/adaptivethreatmodeling/test/generic_feature_testing/TestUtil.qll
+++ b/javascript/ql/experimental/adaptivethreatmodeling/test/generic_feature_testing/TestUtil.qll
@@ -1,16 +1,5 @@
 import javascript
-import experimental.adaptivethreatmodeling.FeaturizationConfig
-
-/**
- * A featurization config that featurizes all endpoints.
- *
- * Ideally this should not be in here, but it is needed for EnclosingFunctionName and EnclosingFunctionBody due to performance considerations :(.
- */
-class NoRestrictionsFeaturizationConfig extends FeaturizationConfig {
-  NoRestrictionsFeaturizationConfig() { this = "NoRestrictionsFeaturization" }
-
-  override DataFlow::Node getAnEndpointToFeaturize() { any() }
-}
+import extraction.NoFeaturizationRestrictionsConfig
 
 class Endpoint extends DataFlow::Node {
   Endpoint() { this.asExpr().(VarAccess).getName() = "endpoint" }