1 files changed, 17 insertions, 3 deletions
diff --git a/swift/test/Ice/proxy/TestAMDI.swift b/swift/test/Ice/proxy/TestAMDI.swift
index 70cd73981d1..58aaaf90012 100644
--- a/swift/test/Ice/proxy/TestAMDI.swift
+++ b/swift/test/Ice/proxy/TestAMDI.swift
@@ -4,7 +4,7 @@
 import Ice
 import PromiseKit
 
-class MyDerivedClassI: MyDerivedClass {
+final class MyDerivedClassI: MyDerivedClass, Object {
     var _ctx: [String: String]
 
     init() {
@@ -27,8 +27,22 @@ class MyDerivedClassI: MyDerivedClass {
         return Promise.value(_ctx)
     }
 
-    func ice_isA(s: String, current: Ice.Current) throws -> Bool {
+    func ice_isA(s: String, current: Ice.Current) -> Bool {
+        _ctx = current.ctx
+        return MyDerivedClassDisp.staticIds.contains(s)
+    }
+
+    func ice_ids(current: Ice.Current) -> [String] {
+        _ctx = current.ctx
+        return MyDerivedClassDisp.staticIds
+    }
+
+    func ice_id(current: Ice.Current) -> String {
+        _ctx = current.ctx
+        return MyDerivedClassDisp.staticId
+    }
+
+    func ice_ping(current: Ice.Current) {
         _ctx = current.ctx
-        return try ice_ids(current: current).contains(s)
     }
 }