fix null check