Fixed errors in model associations