trying to fix generator