diff --git a/mo/jobs/__init__.py b/mo/jobs/__init__.py
index 8110ce5c02c46ec78320e01ffdcc00c0eb3c24af..f007160426064e798d9f8f9bc3c0fae837db8221 100644
--- a/mo/jobs/__init__.py
+++ b/mo/jobs/__init__.py
@@ -87,7 +87,7 @@ class TheJob:
 
         job = self.job
         logger.info(f'Job: Expiruji job #{job.job_id}')
-        assert job.state in (db.JobStatedb.JobState.done, db.JobState.failed)
+        assert job.state in (db.JobState.done, db.JobState.failed)
         return self._finish_remove()
 
     def remove_loaded(self):