Missed a self variable in two places