diff --git a/src/play_clj/g2d_physics.clj b/src/play_clj/g2d_physics.clj
index c2b0cfb..acb8af3 100644
--- a/src/play_clj/g2d_physics.clj
+++ b/src/play_clj/g2d_physics.clj
@@ -268,7 +268,7 @@ such as :on-begin-contact."
   ([screen entities]
     (step! screen)
     (map (fn [e]
-           (if (u/get-obj e :body)
+           (if (:body e)
              (assoc e
                     :x (body-x e)
                     :y (body-y e)
diff --git a/src/play_clj/g3d_physics.clj b/src/play_clj/g3d_physics.clj
index 6afe0a8..615809a 100644
--- a/src/play_clj/g3d_physics.clj
+++ b/src/play_clj/g3d_physics.clj
@@ -318,7 +318,7 @@ such as :on-begin-contact."
   ([screen entities]
     (step! screen)
     (map (fn [e]
-           (if (u/get-obj e :body)
+           (if (:body e)
              (assoc e
                     :x (body-x e)
                     :y (body-y e)