chaned the tests to better test the last function