fixed bug with logger in model_class.py