Fix some more things

Signed-off-by: Sam Phippen <samphippen@googlemail.com>