These last two lines are the only changes; otherwise, just an indentation change and moved above so multiple tests can use it.