Fixed viewmodel bug