diff --git a/sqlalchemy_utils/functions/orm.py b/sqlalchemy_utils/functions/orm.py
index b35d7e2..2135fb3 100644
--- a/sqlalchemy_utils/functions/orm.py
+++ b/sqlalchemy_utils/functions/orm.py
@@ -132,16 +132,21 @@ def query_entities(query):
     Example::
 
 
-        query = session.query(
-            Category
-        )
+        query = session.query(Category)
+
+        query_entities(query)  # <Category>
+
+
+        query = session.query(Category.id)
+
+        query_entities(query)  # <Category>
 
-        query_entities(query)  # ('Category', )
 
     :param query: SQLAlchemy Query object
     """
     for entity in query._entities:
-        yield entity.entity_zero.class_
+        if entity.entity_zero:
+            yield entity.entity_zero.class_
 
     for entity in query._join_entities:
         if isinstance(entity, Mapper):