Categories display bug fix

closes #209 #114