Missed the actual rename of the Generator class.
Part of issue #473