Fixing badly referenced class