Fixed getting wrong model