fix missing import, add a test