fix models :P