Fix null return