diff --git a/data-transformation/Data Transformation CO2 vs Temperature.py b/data-transformation/Data Transformation CO2 vs Temperature.py
index 1b8b221..89bf4d9 100644
--- a/data-transformation/Data Transformation CO2 vs Temperature.py	
+++ b/data-transformation/Data Transformation CO2 vs Temperature.py	
@@ -587,10 +587,10 @@ def fix_country(col: Column) -> Column:
 
 # COMMAND ----------
 
-def test_fix_country(self):
+def test_fix_country():
   original = pd.Series(["  gErMaNy ", "   uNiTeD sTaTeS    "])
-  spark_df = self.spark.createDataFrame(pd.DataFrame({"Country": original}))
-  spark_df = spark_df.withColumn("Country", self.transformer.fix_country(F.col("Country")))
+  spark_df = spark.createDataFrame(pd.DataFrame({"Country": original}))
+  spark_df = spark_df.withColumn("Country", fix_country(F.col("Country")))
   fixed = spark_df.toPandas()
   try:
     result = sorted(fixed["Country"])
@@ -598,6 +598,8 @@ def test_fix_country(self):
   except Exception as e:
     raise type(e)(''.join(debug(original))) from e
   print("All tests pass! :)")
+  
+test_fix_country()
 
 # COMMAND ----------