Fix null returns