Generator Error on test 4