fix spec errors on model return