trying to fix models

May delete this commit