diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index dd9006cae57e667c3e804d2224444241ef4d16af..082617566edb395c282491c04c9d38da8ae2ed16 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -4,10 +4,7 @@ import os
 import numpy as np
 import yaml
 
-from semantic_router.encoders import (
-    BaseEncoder,
-    OpenAIEncoder
-)
+from semantic_router.encoders import BaseEncoder, OpenAIEncoder
 from semantic_router.linear import similarity_matrix, top_scores
 from semantic_router.llms import BaseLLM, OpenAILLM
 from semantic_router.route import Route
@@ -153,6 +150,7 @@ class RouteLayer:
     index: np.ndarray | None = None
     categories: np.ndarray | None = None
     score_threshold: float
+    encoder: BaseEncoder
 
     def __init__(
         self,