Oops, forgot some test code