[Models] add missing import