Fixed non-compiling issue