Model fix

now works