fix variable name
change to fit sklean 0.16 api