Added import...

...to resolve compilation error