Minor Fix

-Categorization bug fix