Fix warning in code.