Updating the test that was using arraylist intead of dictionary