Fixed tests with custom User model.