Fixed minor compiling issue