diff --git a/src/active_grasp/policy.py b/src/active_grasp/policy.py
index 25c334d..0d06996 100644
--- a/src/active_grasp/policy.py
+++ b/src/active_grasp/policy.py
@@ -91,7 +91,7 @@ class Policy:
         return filtered_grasps[i], qualities[i], scores[i]
 
     def score_fn(self, grasp, quality, q, q_grasp):
-        return grasp.pose.translation[2]
+        return quality
 
 
 class SingleViewPolicy(Policy):