Fixes to build borks

Signed-off-by: MassStash <MassStash@gmail.com>