fix compile error

Signed-off-by: Bryan Smith <zarboz@gmail.com>