Bug fixes

Signed-off-by: Tim Van Steenburgh <tvansteenburgh@gmail.com>