diff --git a/src/router_dataset.py b/src/router_dataset.py
index b251f01ad..723a842cc 100644
--- a/src/router_dataset.py
+++ b/src/router_dataset.py
@@ -7,8 +7,6 @@
 from fastapi.responses import ORJSONResponse
 from pydantic import BaseModel, validator
 
-from .signals.semantic_similarity import SemanticSimilaritySignal
-
 from .config import data_path
 from .data.dataset import BinaryOp
 from .data.dataset import Column as DBColumn
@@ -31,6 +29,7 @@
 from .signals.concept_labels import ConceptLabelsSignal
 from .signals.concept_scorer import ConceptScoreSignal
 from .signals.default_signals import register_default_signals
+from .signals.semantic_similarity import SemanticSimilaritySignal
 from .signals.signal import (
   Signal,
   TextEmbeddingModelSignal,