fix model :(