fix category bug :smile: