Modify the code that now the CBS is correct.