Small fixes

Signed-off-by: Dries Vints <dries.vints@gmail.com>