Fixed missing import on first install