Updating unit test