diff --git a/Makefile b/Makefile
index beae14e..b971207 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,6 @@
 gen:
-	rm -rf schemas
-	python generator.py queues
-	python -m grpc_tools.protoc --proto_path schemas --python_out=. --pyi_out=. --grpc_python_out=. ./schemas/queues/tasks.proto
-	rm -rf schemas
+	curl https://platform.sprinthub.ru/generator >> generator.py
+	python generator.py
+	rm generator.py
 run:
 	python ./server.py
\ No newline at end of file
diff --git a/generator.py b/generator.py
deleted file mode 100644
index 2da229c..0000000
--- a/generator.py
+++ /dev/null
@@ -1,16 +0,0 @@
-import json
-import urllib.request
-import sys
-import os
-
-
-arg = sys.argv[-1]
-
-
-response = urllib.request.urlopen(f'https://platform.sprinthub.ru/schemas/get?project={arg}').read()
-data = json.loads(response)
-os.mkdir('schemas')
-os.mkdir(f'schemas/{arg}')
-for key, value in data.items():
-    with open(f'schemas/{arg}/{key}', 'w+') as fp:
-        fp.write(value)
diff --git a/server.py b/server.py
index 5496f66..6291e8d 100644
--- a/server.py
+++ b/server.py
@@ -3,8 +3,8 @@ import datetime
 import grpc
 import bson
 
-import tasks_pb2
-import tasks_pb2_grpc
+from queues import tasks_pb2
+from queues import tasks_pb2_grpc
 
 from utils import time
 from storage.mongo import tasks