Generator fix

fixes generator error.