fix to models