Update tests (new custom User model)