Fix compile error caused by commit 6fd17c43