TST: Stupid missing import