diff --git a/evoprompt/api/backend.py b/evoprompt/api/backend.py
index 0595f7aecfe0d863299b6567d0a0888ccf63a7a0..608d26bd2f4aa634040836bdb67d7fe33dff82de 100644
--- a/evoprompt/api/backend.py
+++ b/evoprompt/api/backend.py
@@ -10,7 +10,7 @@ from evoprompt.api import config
 from evoprompt.cli import argument_parser
 from evoprompt.evolution import GeneticAlgorithm
 from evoprompt.models import Llama, LLMModel
-from evoprompt.task import SentimentAnalysis
+from evoprompt.task.sentiment_analysis import SST2
 
 
 class MultiProcessOptimizer:
@@ -34,7 +34,7 @@ class MultiProcessOptimizer:
             [
                 "--debug" if debug else "",
                 "--task",
-                SentimentAnalysis.shorthand,
+                SST2.shorthand,
             ]
         )
 
@@ -55,7 +55,7 @@ class MultiProcessOptimizer:
         ).result()
 
         # currently fix task
-        task = SentimentAnalysis(evaluation_model, self.options)
+        task = SST2(evaluation_model, **vars(self.options))
 
         optimizer_class = GeneticAlgorithm
         # optimizer_class = DifferentialEvolution