Compile warning fixed