Storageで生成されたURLのFireStoreへの保存

Firebase Auth/Storage/FireStore を使った新規登録の処理を作っています。

以下は新規登録(まずAuthで認証し、帰ってきたUserをFireStoreに登録、画像があるならばそのURLも一緒に保存する)処理です。

static func signUp(name: String, email: String, password: String, image: UIImage?, onSuccess: @escaping () -> Void, onError: @escaping (_ errorMessage: String?) -> Void) {
        Auth.auth().createUser(withEmail: email, password: password, completion: { user, error in
            if error != nil {
                onError(error)
                return
            }
            guard let uid = user?.user.uid else { return }
            var dict: [String: Any] = [
                "name": name,
                "email": email
            ]
            // If Image is Set
            if let image = image {
                StorageService.storage(image: image, path: .icon, id: uid) { (imageUrl) in
                    dict["iconUrl"] = imageUrl
                }
            }
            Firestore.firestore().collection("users").document(uid).setData(dict) { (error) in
                if let error = error {
                    print(error)
                    return
                }
            }
            onSuccess()
        })
    }

以下はStorageのUIImageを引数にとり、URLを返す処理を関数にしたものです

class StorageService {

    // Upload Image to Storage
    static func storage(image: UIImage?, path: PathType, id: String, completion: @escaping (_ imageUrl: String?) -> ()) {
        guard let image = image, let imageData = UIImageJPEGRepresentation(image, 0.1) else {
            print("Non Image")
            completion(nil)
            return
        }
        let storageRef = Storage.storage().reference().child(path.rawValue).child(id)
        storageRef.putData(imageData, metadata: nil, completion: { (metaData, error) in
            if let error = error {
                print("Fail to Put Data in Storage : \(error)")
                completion(nil)
                return
            }
            storageRef.downloadURL { (imageUrl, error) in
                if let error = error {
                    print("Fail to Download Url : \(error)")
                    completion(nil)
                    return
                }
                if let imageUrl = imageUrl?.absoluteString {
                    completion(imageUrl)
                }
            }
        })
    }
}

Authの登録と、FireStoreへの保存は成功するのですが、画像がある時、
Storageには保存されているのに、FireStoreには画像のURlが保存されません。

storage()クロージャの書き方などに問題があるのでしょうか？