Minor code cleanup

Codefix.