Hacky fix to make Model compile