Fix category feature