Fix build error

This pull request fix bulid error that introduced by 1db2dcb824