Minor bug fix in cp module