Updated my unit test