Fix compilation error.

There was still a dependency on a removed class.