bug fixes

Signed-off-by: Kyle Noland <kylenoland@gmail.com>