diff --git a/test/integration/test_data_observatory_functions.py b/test/integration/test_data_observatory_functions.py
index 0151f7b..8955d50 100644
--- a/test/integration/test_data_observatory_functions.py
+++ b/test/integration/test_data_observatory_functions.py
@@ -15,9 +15,9 @@ class TestDataObservatoryFunctions(TestCase):
         )
 
     def test_if_get_demographic_snapshot_is_ok(self):
-        query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669));&api_key={0}".format(self.env_variables['api_key'])
-        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
-        assert_not_equal(routing['the_geom'], None)
+        query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669)) as snapshot;&api_key={0}".format(self.env_variables['api_key'])
+        snapshot = IntegrationTestHelper.execute_query(self.sql_api_url, query)
+        assert_not_equal(snapshot['snapshot'], None)
 
     def test_if_get_demographic_snapshot_without_api_key_raise_error(self):
         query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669));"
@@ -27,9 +27,9 @@ class TestDataObservatoryFunctions(TestCase):
             assert_equal(e.message[0], "The api_key must be provided")
 
     def test_if_get_segment_snapshot_is_ok(self):
-        query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669));&api_key={0}".format(self.env_variables['api_key'])
-        routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
-        assert_not_equal(routing['the_geom'], None)
+        query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669)) as snapshot;&api_key={0}".format(self.env_variables['api_key'])
+        snapshot = IntegrationTestHelper.execute_query(self.sql_api_url, query)
+        assert_not_equal(snapshot['snapshot'], None)
 
     def test_if_get_segment_snapshot_without_api_key_raise_error(self):
         query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669));"