Add missing class.

This got missed during a Groovy to Java conversion.