And don't forget to remove the test code