Update tests for cyclomatic complexity