Fix build

Signed-off-by: Paul Beeler <sparksco@gmail.com>