Add missing time import