Fixing missing models issue