Fixing auto loading of correct viewmodel