Fix missing import condition