small changes needed to compile