diff --git a/src/dq_suite/validation.py b/src/dq_suite/validation.py
index e763f40..aa18027 100644
--- a/src/dq_suite/validation.py
+++ b/src/dq_suite/validation.py
@@ -140,12 +140,11 @@ def _get_gx_expectation_object(validation_rule: Rule, table_name: str):
         gx_expectation_class = getattr(gx_core, gx_expectation_name)
 
         gx_expectation_parameters: dict = validation_rule["parameters"]
-        if "column" not in gx_expectation_parameters.keys():
-            gx_expectation_parameters["column"] = None
+        column_name = gx_expectation_parameters.get("column", None)
 
         gx_expectation_parameters["meta"] = {
             "table_name": table_name,
-            "column_name": gx_expectation_parameters['column'],
+            "column_name": column_name,
             "expectation_name": gx_expectation_name
         }
         return gx_expectation_class(**gx_expectation_parameters)