Fixed CS, added null check.