Fix: docblock error
Fix: remove unused class