fix: remove class label