diff --git a/pistol_magazine/self_made.py b/pistol_magazine/self_made.py
index 7044e33..dad8441 100644
--- a/pistol_magazine/self_made.py
+++ b/pistol_magazine/self_made.py
@@ -102,37 +102,6 @@ def model_to_data(cls, data: dict or list) -> Dict or List:
     def get_datatype(self):
         return self.models.get_datatype()
 
-    # def mock(
-    #         self, to_json: bool = False,
-    #         num_entries: Optional[int] = None,
-    #         key_generator: Optional[Callable[[], str]] = None,
-    #         as_list: bool = False
-    # ):
-    #     if key_generator is None:
-    #         key_generator = lambda: str(uuid.uuid4())
-    #     hook_manager.trigger_hooks('pre_generate', None)
-    #     if num_entries is not None:
-    #         if as_list:
-    #             final_result = []
-    #             for _ in range(num_entries):
-    #                 data = self.models.mock(to_json=False)
-    #                 data = hook_manager.trigger_hooks('after_generate', data)
-    #                 final_result.append(data)
-    #         else:
-    #             final_result = {}
-    #             for _ in range(num_entries):
-    #                 entry_key = key_generator()
-    #                 data = self.models.mock(to_json=False)
-    #                 data = hook_manager.trigger_hooks('after_generate', data)
-    #                 final_result[entry_key] = data
-    #         result = final_result
-    #     else:
-    #         result = self.models.mock(to_json=False)
-    #         result = hook_manager.trigger_hooks('after_generate', result)
-    #     if to_json:
-    #         result = json.dumps(result)
-    #     result = hook_manager.trigger_hooks('final_generate', result)
-    #     return result
     def mock(
             self, to_json: bool = False,
             num_entries: Optional[int] = None,
diff --git a/tests/test_selfmade.py b/tests/test_selfmade.py
index 04d170e..5d0e6e1 100644
--- a/tests/test_selfmade.py
+++ b/tests/test_selfmade.py
@@ -1,8 +1,7 @@
 from pprint import pprint
-from random import choice
-from pistol_magazine import DataMocker, provider, Str, Int, Timestamp, Bool, Dict, Float, List, Datetime, StrInt
+from random import choice, randint
+from pistol_magazine import *
 from pistol_magazine.hooks.hooks import hook
-from pistol_magazine.self_made import ProviderField
 
 
 def test_model_data_conversion():