Add another missing import