[ObjC] Set return type on objc_msgSend calls that send an init message to a known class

1 files changed, 1 insertions, 2 deletions

diff --git a/plugins/workflow_objc/src/activities/super_init.rs b/plugins/workflow_objc/src/activities/super_init.rs
index 81b9a453..7b6bb3f7 100644
--- a/plugins/workflow_objc/src/activities/super_init.rs
+++ b/plugins/workflow_objc/src/activities/super_init.rs
@@ -39,8 +39,7 @@ fn return_type_for_super_init(call: &util::Call, view: &BinaryView) -> Option<Re
         util::match_constant_pointer_or_load_of_constant_pointer(&call.call.params[1])?;
     let selector = Selector::from_address(view, selector_addr).ok()?;
 
-    // TODO: This will match `initialize` and `initiate` which are not init methods.
-    if !selector.name.starts_with("init") {
+    if !selector.is_init_family() {
         return None;
     }