Fix tests for install_generator