Fix build error
Signed-off-by: Kristofer Ard <kristoferard@gmail.com>