Fix mismatched variable name
Follow to changes of the implementation