minor: adding missing import