Fix unit test for api change
Add generator to dockerfile