Fixed model names in relations