Mini fix
Removed incomplete generator