diff --git a/examples/glvq_spiral.py b/examples/glvq_spiral.py
index b9b34f6..eb0adcb 100644
--- a/examples/glvq_spiral.py
+++ b/examples/glvq_spiral.py
@@ -4,8 +4,7 @@ import argparse
 
 import prototorch as pt
 import pytorch_lightning as pl
-
-#import torch
+import torch
 
 if __name__ == "__main__":
     # Command-line arguments