Fix class name to match formula name change