diff --git a/build.sbt b/build.sbt
index 154d7d96..f77a08d7 100644
--- a/build.sbt
+++ b/build.sbt
@@ -262,6 +262,10 @@ lazy val datasetSettings =
         imt("frameless.RecordEncoderFields.deriveRecordLast"),
         mc("frameless.functions.FramelessLit"),
         mc(f"frameless.functions.FramelessLit$$"),
+        mc("org.apache.spark.sql.FramelessInternals"),
+        mc("org.apache.spark.sql.FramelessInternals$"),
+        mc("org.apache.spark.ml.FramelessInternals"),
+        mc("org.apache.spark.ml.FramelessInternals$"),
         mc("frameless.MapGroups"),
         mc(f"frameless.MapGroups$$"),
         dmm("frameless.functions.package.litAggr"),