Forgot to add the test to the initial commit