Fixed incorrect model integration