fix: remove wrong class