Fix null bug