diff --git a/README.md b/README.md
index 4908f1d..1e5dcd3 100644
--- a/README.md
+++ b/README.md
@@ -7,6 +7,8 @@ pip install git+https://github.com/graphcore-research/pytorch-tensor-tracker
 ```
 
 ```python
+import tensor_tracker
+
 with tensor_tracker.track(module) as tracker:
     module(inputs).backward()