Fixes test broken by fixing #23.

Signed-off-by: James McCarthy <james@thisishatch.co.uk>