Fix compile error caused by commit 9a288e8