fix: modified class