Another cleanup

Signed-off-by: Dom Pascal <dom@dompascal.com>