Added necessary import

[none]