Fixed compile error caused by latest modification