Update test for class rename