bug fix

philipithomas · philipithomas · commit 6eaec6d1f8db · 2016-02-23T11:05:24.000-08:00
diff --git a/staffjoy/resource.py b/staffjoy/resource.py
@@ -170,7 +170,7 @@ def create(cls, parent=None, **kwargs):
 
         # No envelope on post requests
         data = response.json()
-        obj.route[obj.ID_NAME] = data.get("id", obj.ID_NAME)
+        obj.route[obj.ID_NAME] = data.get("id", data.get(obj.ID_NAME))
         obj.data = data
 
         return obj
