Forgot to update tests before last commit