oops fix generator stuff