Make sure test returns correct string

Signed-off-by: Ben Agricola <bagricola@squiz.co.uk>