diff --git a/README.md b/README.md
index e02d4a6..9dd5fd1 100644
--- a/README.md
+++ b/README.md
@@ -57,6 +57,7 @@ Here is a minimal example assuming you have a pointcloud in the folder data.
 ```python
 
 from mask3d import get_model, load_mesh, prepare_data, map_output_to_pointcloud, save_colorized_mesh 
+import torch
 
 model = get_model('checkpoints/scannet200/scannet200_benchmark.ckpt')
 model.eval()