fix install error

Signed-off-by: Wang Zhen <yuchen12@gmail.com>