More clean up

Signed-off-by: Richard Bagshaw <richard@bagshaw.co.uk>