diff --git a/dremio_client/flight/__init__.py b/dremio_client/flight/__init__.py
index 4d7a39b..b93b0e0 100644
--- a/dremio_client/flight/__init__.py
+++ b/dremio_client/flight/__init__.py
@@ -146,11 +146,9 @@ def query(
                 batches.append(batch)
             except StopIteration:
                 break
-        data = pa.Table.from_batches(batches)
-        if pandas:
-            return data.to_pandas()
-        else:
-            return data
+        table = pa.Table.from_batches(batches)
+        
+        return table
 
 
 except ImportError: