Fixed missing import. Thank you dynamic languages