That's true, but it feels really hacky to pass `class_name`