diff --git a/core/pom.xml b/core/pom.xml
index a1bdd8ec68aeb..d87e2bca030e3 100644
--- a/core/pom.xml
+++ b/core/pom.xml
@@ -266,6 +266,11 @@
junit-interface
test
+
+ org.spark-project
+ pyrolite
+ 2.0
+
target/scala-${scala.binary.version}/classes
diff --git a/pom.xml b/pom.xml
index c84829dcf23b8..616563809e97b 100644
--- a/pom.xml
+++ b/pom.xml
@@ -264,11 +264,6 @@
stream
2.5.1
-
- org.spark-project
- pyrolite
- 2.0
-