add missing import, fixes #3