Fix missing import - Ref #299