From 4eb133ce174982fc88e3398e7cbba1cba390115a Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Tue, 7 Jul 2026 10:01:59 +0200
Subject: [PATCH] fix(checker): handle setting unknown as column

---
 midas/checker/frames/column_methods.py | 2 +-
 midas/checker/frames/frame_manager.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/midas/checker/frames/column_methods.py b/midas/checker/frames/column_methods.py
index 345a884..d370398 100644
--- a/midas/checker/frames/column_methods.py
+++ b/midas/checker/frames/column_methods.py
@@ -117,7 +117,7 @@ class ColumnMethodRegistry(MethodRegistry[Call]):
         # Build signature with new column type and generic operand
         param_type: TypeVar = TypeVar(name="T", bound=None)
         signature = GenericType(
-            name="add",
+            name=method,
             params=[param_type],
             body=Function(
                 params=ParamSpec(
diff --git a/midas/checker/frames/frame_manager.py b/midas/checker/frames/frame_manager.py
index ca75187..74db171 100644
--- a/midas/checker/frames/frame_manager.py
+++ b/midas/checker/frames/frame_manager.py
@@ -132,7 +132,7 @@ class FrameManager:
                 location,
                 f"Cannot assign {type} to dataframe column. Must be a ColumnType",
             )
-            return frame
+            return self._set_column(frame, name, ColumnType(type=UnknownType()))
         return self._set_column(frame, name, type)
 
     def get(