added missing class for latest changes