diff --git a/tools/wavegen/Makefile b/tools/wavegen/Makefile
index ce2d6c72e8..9703911cf9 100644
--- a/tools/wavegen/Makefile
+++ b/tools/wavegen/Makefile
@@ -8,7 +8,9 @@ setup:
 	npm ci
 
 run:
-	npx ts-node src/wavegen.ts ../../ui/src ../../py/h2o_wave ../../r/R
+	npx ts-node src/wavegen.ts ../../ui/src ../../py/h2o_wave/h2o_wave ../../r/R
+	cp ../../py/h2o_wave/h2o_wave/ui.py ../../py/h2o_wavelite/h2o_wavelite
+	cp ../../py/h2o_wave/h2o_wave/types.py ../../py/h2o_wavelite/h2o_wavelite
 
 clean:
 	rm -rf build node_modules