Fixing Mongo Model issue