More minor bug fixes.

Signed-off-by: Dave Bush <davebush@gmail.com>