Code example fix

Fixed missing import.