diff --git a/pykos/Makefile b/pykos/Makefile
index f47ff64..d5c35f9 100644
--- a/pykos/Makefile
+++ b/pykos/Makefile
@@ -38,6 +38,10 @@ generate-proto:
 	mv kos/* kos_protos/
 	rm -rf kos
 	touch kos_protos/__init__.py
+
+	# Fix imports in all generated files
+	find kos_protos -type f -name "*.py" -exec sed -i '' 's/from kos/from kos_protos/g' {} +
+	find kos_protos -type f -name "*.py" -exec sed -i '' 's/import kos/import kos_protos/g' {} +
 .PHONY: generate-proto