fix hardcoded generator