clean up shit in generator test case