diff --git a/src/model/model.py b/src/model/model.py
index 6e43f6d..7a56bef 100644
--- a/src/model/model.py
+++ b/src/model/model.py
@@ -1,6 +1,8 @@
 import os
 from abc import ABC, abstractmethod
 
+from IPython.utils.wildcard import is_type
+
 from rbm.entity import Entity
 from rbm.train import train
 from rbm.matrix import Mat, np
@@ -18,6 +20,8 @@ class Model(ABC):
 		for name, value in self.__dict__.items():
 			if isinstance(value, obj_type):
 				obj_list.append(value)
+			elif isinstance(value, type(obj_list)):
+				obj_list += value
 		return obj_list
 
 	def train(self, batch: Mat, status: Status = None):