Fixing build issue

Signed-off-by: Adam Jimerson <vendion@gmail.com>