updating the test to comply to the new code