(#6): Fix KeyError too for the modification object part