fix yhat autograding demo