Make accessing the ID easer

1 files changed, 2 insertions, 2 deletions

diff --git a/web-server/opendc/models/simulation.py b/web-server/opendc/models/simulation.py
index bf19368c..dbe1e800 100644
--- a/web-server/opendc/models/simulation.py
+++ b/web-server/opendc/models/simulation.py
@@ -18,7 +18,7 @@ class Simulation(Model):
         """
         user = User.from_google_id(google_id)
         authorizations = list(
-            filter(lambda x: str(x['simulationId']) == str(self.obj['_id']), user.obj['authorizations']))
+            filter(lambda x: str(x['simulationId']) == str(self.get_id()), user.obj['authorizations']))
         if len(authorizations) == 0 or (edit_access and authorizations[0]['authorizationLevel'] == 'VIEW'):
             raise ClientError(Response(403, "Forbidden from retrieving simulation."))
 
@@ -26,6 +26,6 @@ class Simulation(Model):
         """Get all user IDs having access to this simulation."""
         return [
             user['_id'] for user in DB.fetch_all({'authorizations': {
-                'simulationId': self.obj['_id']
+                'simulationId': self.get_id()
             }}, User.collection_name)
         ]