diff --git a/src/user_interface.py b/src/user_interface.py
index bb23e39..9ea72d5 100644
--- a/src/user_interface.py
+++ b/src/user_interface.py
@@ -1,11 +1,10 @@
-from pydantic import BaseModel
+from dataclasses import dataclass
 from secrets import choice
-
 from src.models import KeypadSize
 from src.utils import list_to_matrix, secure_fisher_yates_shuffle, matrix_to_list, matrix_transpose
 
-
-class UserInterface(BaseModel):
+@dataclass
+class UserInterface:
     interface: list[int]
     keypad_size: KeypadSize