diff --git a/tests/debug.py b/tests/debug.py
index 42b5ef75..3f0c58f0 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -8,6 +8,8 @@ import warnings
 import torch
 import os
 import shutil
+import pdb
+
 
 def get_model():
     # set up model with these hyperparams