Fix null object bug