Add missing import, explains why it was broken before