fix: change class to object