Fixed removeCategory a little bit