diff --git a/tests/integration/resources/test-iceberg.py b/tests/integration/resources/test-iceberg.py
index 4675ed87..b4be2551 100644
--- a/tests/integration/resources/test-iceberg.py
+++ b/tests/integration/resources/test-iceberg.py
@@ -9,16 +9,12 @@
 args = parser.parse_args()
 num_rows = args.num_rows
 
-spark = SparkSession\
-    .builder\
-    .appName("IcebergExample")\
-    .getOrCreate()
+spark = SparkSession.builder.appName("IcebergExample").getOrCreate()
 
 
-schema = StructType([
-  StructField("row_id", LongType(), True),
-  StructField("row_val", LongType(), True)
-])
+schema = StructType(
+    [StructField("row_id", LongType(), True), StructField("row_val", LongType(), True)]
+)
 
 data = []
 for idx in range(num_rows):
@@ -31,4 +27,4 @@
 
 df = spark.table("demo.foo.bar")
 count = df.count()
-print(f"Number of rows inserted: {count}")
\ No newline at end of file
+print(f"Number of rows inserted: {count}")