Fixed category removement bug