diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index cea08bbc3b..b041277e47 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -16,6 +16,7 @@ __all__ = [
     "PatchRL",
 ]
 
+import torch
 
 def PatchRL(FastLanguageModel):