Fix/Add null checkcs