Fix import leftover