Fixes data_class attribute