Clean up generator