fixed issue with model