adjusting the code of the first test a little