tidied up test to make it clearer