Fix a couple of minor bugs.

Signed-off-by: Peter Chester <peter@tri.be>