diff --git a/crates/ml-alpha/cuda/rl_position_heat_check.cu b/crates/ml-alpha/cuda/rl_position_heat_check.cu
index 58f4e35dc..aa029d4ea 100644
--- a/crates/ml-alpha/cuda/rl_position_heat_check.cu
+++ b/crates/ml-alpha/cuda/rl_position_heat_check.cu
@@ -52,10 +52,10 @@ extern "C" __global__ void rl_position_heat_check(
 
     if (b < b_size && cap > 0) {
         const int position_lots = *(const int*)(pos_state + b * pos_bytes);
-        if (position_lots > cap) {
+        if (position_lots >= cap) {
             actions[b] = ACTION_FLAT_FROM_LONG;
             s_flags[tid] = 1;
-        } else if (position_lots < -cap) {
+        } else if (position_lots <= -cap) {
             actions[b] = ACTION_FLAT_FROM_SHORT;
             s_flags[tid] = 1;
         }