Why did Avatar Korra still hallucinate her battle with Zaheer even though all the poison was out?

As mentioned by Kadima in the comments - Korra was still suffering from the trauma of her battle with Zaheer. Getting the poison out helped her physical abilities to a point, but Korra still needed to mentally come to terms with her loss against Zaheer before being able to bend like she used to.