minor class fix