Update CPS to fix small regression