Fix minor errors in models