Fix test that I broke

We probably need a better way of handling this than just stubbing things
constantly. Should `GenericEdition` be in the list of allowed classes?