fixed category bug
small change