Fix a test after change on logic for NumberGenerator