Small one line change, made figure_class field a local method variable