Fix reference to non-existant class