Updating our assertion.
Adding test for generator type.