diff --git a/Makefile b/Makefile
index 67fee6d..49b877a 100644
--- a/Makefile
+++ b/Makefile
@@ -12,7 +12,7 @@ profile:
 	nsys profile --capture-range=cudaProfilerApi --cuda-graph-trace=node --capture-range-end=stop -o profile -f true python train.py
 
 run:
-	./build/inference export/model.so
+	./build/inference ${PWD}/export/model.so
 
 .PHONY: run nsys build clean
 
diff --git a/README.md b/README.md
index e3f7ae0..326218b 100644
--- a/README.md
+++ b/README.md
@@ -1,8 +1,7 @@
 ## TODO
 
-- [x] Get `torch.export` pipeline working.
-- [ ] Train tetris
-- [ ] Port `allegro-jax`
-
+- [x] Train tetris
+- [x] Get `torch.export` pipeline working
+- [] Call mode.so successfully in cpp