Fix 017b9eb bugs