Fixed not-compiling bug