Fix for CFG unit-test.

Test is checking the wrong function.