Fix another missing import

This is why using an IDE is a Good Plan, kids.