Write a test for it too