fixed an issue with model design