fixed Generator unit tests for new method signature