diff --git a/worker-{{cookiecutter.slug}}/worker_{{cookiecutter.slug}}/worker.py b/worker-{{cookiecutter.slug}}/worker_{{cookiecutter.slug}}/worker.py
index 6f7c7c02210618035bd8731834561ccc6db83d1c..16219c4b60e4aec8312a1c0fcda08cf9412a0286 100644
--- a/worker-{{cookiecutter.slug}}/worker_{{cookiecutter.slug}}/worker.py
+++ b/worker-{{cookiecutter.slug}}/worker_{{cookiecutter.slug}}/worker.py
@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
-from arkindex_worker.worker import BaseWorker
+from arkindex_worker.worker import ElementsWorker
 
 
-class Demo(BaseWorker):
+class Demo(ElementsWorker):
     def process_element(self, element):
         print("Demo processing element", element)