Fix correlation count not updating