Issue #13. Fixed wrong model