Minor fix of return promise.