And add a test