Modified getCategory to return null if Integer is null