add missing import
fix comment on result type