Fix build

Signed-off-by: Paul Beeler <pbeeler80@gmail.com>
Fix build #2