You haven't implemented __eq__ so this does the wrong thing I think