Fix setGenerator()