Wither boss still flying

Wither boss may still fly even when affected by wither armor after going down to 50% hp, it may also get stuck in the air (not stuck as in not moving at all but rather it's not coming close enough to the ground to hit it)