fix removal of 1 correlation values

Signed-off-by: Zarrar Shehzad <czarrar@gmail.com>