Fixing missing model in files