Forgot to add a test