Fix category problem