diff --git a/src/csrc/rl/off_policy/interface.cpp b/src/csrc/rl/off_policy/interface.cpp
index d7b5e6f..967dee5 100644
--- a/src/csrc/rl/off_policy/interface.cpp
+++ b/src/csrc/rl/off_policy/interface.cpp
@@ -183,7 +183,6 @@ torchfort_result_t torchfort_rl_off_policy_train_step(const char* name, float* p
   c10::cuda::OptionalCUDAStreamGuard stream_guard;
   c10::cuda::OptionalCUDAGuard cuda_guard;
   set_device_and_stream(stream_guard, cuda_guard, model_device, ext_stream);
-  set_device_and_stream(stream_guard, cuda_guard, rb_device, ext_stream);
 #endif
 
   try {