Added a missing import to resolve compilation.