fix build

Signed-off-by: cj360 <ayunker551@gmail.com>