Fixed issue with Model Size