Clean up the code: fix warning.