Fixed the bug that prevented compiling