fix for kmeans state updater - NPE. Pass correct fields