Rename test class to match class renaming