Fix generator tests that were failing due to a stronger assert