Avoid hardcoding the class name