diff --git a/task_geo/testing/__init__.py b/task_geo/testing/__init__.py
index c9a0ba8..5af66e6 100644
--- a/task_geo/testing/__init__.py
+++ b/task_geo/testing/__init__.py
@@ -27,12 +27,12 @@ def check_dataset_format(data):
     if granularity is not None:
         min_granularity_term = LOCATION_TERMS[granularity]
 
-    locations = [data.columns.get_loc(LOCATION_TERMS[granularity])]
-    for term in LOCATION_TERMS[granularity + 1:]:
-        locations.append(check_column_and_get_index(term, data, min_granularity_term))
+        locations = [data.columns.get_loc(LOCATION_TERMS[granularity])]
+        for term in LOCATION_TERMS[granularity + 1:]:
+            locations.append(check_column_and_get_index(term, data, min_granularity_term))
 
-    message = 'The correct ordening of the columns is "country, region, sub_region, city"'
-    assert (locations[::-1] == sorted(locations)), message
+        message = 'The correct ordening of the columns is "country, region, sub_region, city"'
+        assert (locations[::-1] == sorted(locations)), message
 
     message = 'date and timestamp columns should be of datetime dtype'
     time_index = None