diff --git a/src/test/scala/com/crealytics/spark/excel/MaxRowsReadSuite.scala b/src/test/scala/com/crealytics/spark/excel/MaxRowsReadSuite.scala
index 1080489b..26ac33cb 100644
--- a/src/test/scala/com/crealytics/spark/excel/MaxRowsReadSuite.scala
+++ b/src/test/scala/com/crealytics/spark/excel/MaxRowsReadSuite.scala
@@ -16,7 +16,7 @@
 
 package com.crealytics.spark.excel
 
-import com.holdenkarau.spark.testing.DataFrameSuiteBase
+import com.crealytics.spark.DataFrameSuiteBase
 import org.scalatest.matchers.should.Matchers
 import org.scalatest.wordspec.AnyWordSpec
 
diff --git a/src/test/scala/com/crealytics/spark/v2/excel/DataFrameWriterApiComplianceSuite.scala b/src/test/scala/com/crealytics/spark/v2/excel/DataFrameWriterApiComplianceSuite.scala
index 7139ff1e..dc9bb46a 100644
--- a/src/test/scala/com/crealytics/spark/v2/excel/DataFrameWriterApiComplianceSuite.scala
+++ b/src/test/scala/com/crealytics/spark/v2/excel/DataFrameWriterApiComplianceSuite.scala
@@ -125,7 +125,8 @@ class DataFrameWriterApiComplianceSuite extends AnyWordSpec with DataFrameSuiteB
           .save()
 
         val orderedSchemaColumns = dfCsv.schema.fields.map(f => f.name).sorted
-        val expectedDf = dfCsv.union(dfCsv).select(orderedSchemaColumns.head, orderedSchemaColumns.tail.toIndexedSeq: _*)
+        val expectedDf =
+          dfCsv.union(dfCsv).select(orderedSchemaColumns.head, orderedSchemaColumns.tail.toIndexedSeq: _*)
 
         assertWrittenExcelData(expectedDf, targetDir)
       }
diff --git a/src/test/scala/com/crealytics/spark/v2/excel/MaxRowsReadSuite.scala b/src/test/scala/com/crealytics/spark/v2/excel/MaxRowsReadSuite.scala
index 701af94d..ea8ef165 100644
--- a/src/test/scala/com/crealytics/spark/v2/excel/MaxRowsReadSuite.scala
+++ b/src/test/scala/com/crealytics/spark/v2/excel/MaxRowsReadSuite.scala
@@ -16,7 +16,7 @@
 
 package com.crealytics.spark.v2.excel
 
-import com.holdenkarau.spark.testing.DataFrameSuiteBase
+import com.crealytics.spark.DataFrameSuiteBase
 import org.scalatest.matchers.should.Matchers
 import org.scalatest.wordspec.AnyWordSpec