diff --git a/tests/test_decorator.py b/tests/test_decorator.py
index 84630ac..bb17499 100644
--- a/tests/test_decorator.py
+++ b/tests/test_decorator.py
@@ -4,8 +4,7 @@
 SparkSession.builder.getOrCreate()  # Needed for creating the Spark
 
 # Now import the decorators
-from pysparky.decorator import (extension_enabler,
-                                pyspark_column_or_name_enabler)
+from pysparky.decorator import extension_enabler, pyspark_column_or_name_enabler
 
 
 def test_pyspark_column_or_name_enabler():