diff --git a/_modules/postgres_ext.py b/_modules/postgres_ext.py
index 89666fa..3165487 100644
--- a/_modules/postgres_ext.py
+++ b/_modules/postgres_ext.py
@@ -16,7 +16,7 @@ def __virtual__():
     '''
     Only load this module if the postgres module is already loaded
     '''
-    if 'postgres.psql_query' in __salt__:
+    if all((salt.utils.which('psql'), HAS_ALL_IMPORTS)):
         return True
     return False