remove test, add generator