Fix the nullability annotation