Fix import that was causing warning