Fix missing class in magic trait