Whoops, forgot to test that