Forgot to return updated object