Fix a bug introduced by 34482a8 by fixing a missing import