Closes #26 Fix typo for model registration