Null check when setting comment badge