Bugfix

wanted name and was returning whole model