Fix compilation error.

Signed-off-by: johnmave126 <johnmave126@gmail.com>