fix working model bug