fixes

Signed-off-by: Matthew Wilkinson <mat.wilkinson@me.com>