Whoops, forgot to add the test