diff --git a/requirements-dev.txt b/requirements-dev.txt
index efeb72b..86c2fd8 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -2,6 +2,7 @@
 nose2
 coverage
 codeclimate-test-reporter
+greenlet>=3.1
 tox
 mypy==1.1.1
 sqlalchemy[mypy] >= 2.0
diff --git a/sqlalchemy_mixins/activerecord.py b/sqlalchemy_mixins/activerecord.py
index 122a60d..d0e6be5 100644
--- a/sqlalchemy_mixins/activerecord.py
+++ b/sqlalchemy_mixins/activerecord.py
@@ -77,18 +77,18 @@ def destroy(cls, *ids, commit=True):
 
     @classmethod
     def all(cls):
-        return cls.query.all()
+        return cls.session.query(cls).all()
 
     @classmethod
     def first(cls):
-        return cls.query.first()
+        return cls.session.query(cls).first()
 
     @classmethod
     def find(cls, id_):
         """Find record by the id
         :param id_: the primary key
         """
-        return cls.query.get(id_)
+        return cls.session.query(cls).get(id_)
 
     @classmethod
     def find_or_fail(cls, id_):