Fix for compilation warning