diff --git a/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py b/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py
index 4791ff0..4e23f35 100644
--- a/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py
+++ b/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py
@@ -34,8 +34,8 @@ def setup(self) -> None:
         self.desired_score = 1
 
         self.features = {
-            'wafer_count': {'type': 'ordinal'},
-            'score': {'type': 'ordinal'},
+            'wafer_count': {'type': 'ordinal', 'data_type': 'number'},
+            'score': {'type': 'ordinal', 'data_type': 'number'},
             'action': {
                 'type': 'nominal',
                 'data_type': 'number',