added missing test class from last commit