Fix compile error from commit: b555b96acb