Refactor tests to have initialise method