Bugfixed the model