Fix spec test.

- Change class name in the test.