add correlation fix