1 files changed, 4 insertions, 3 deletions
diff --git a/mingling_core/src/program/hook.rs b/mingling_core/src/program/hook.rs
index 56d8e0e..48f632f 100644
--- a/mingling_core/src/program/hook.rs
+++ b/mingling_core/src/program/hook.rs
@@ -144,8 +144,9 @@ where
 {
     /// Adds a typed hook to the program. The hook will be called at the appropriate
     /// lifecycle events.
-    pub fn with_hook(&mut self, hook: ProgramHook<C>) {
+    pub fn with_hook(&mut self, hook: ProgramHook<C>) -> &mut Self {
         self.hooks.push(hook);
+        self
     }
 
     pub(crate) fn run_hook_on_begin(&self, info: HookBeginInfo) {
@@ -678,7 +679,7 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::Groupped;
+    use crate::Grouped;
     use std::sync::atomic::{AtomicBool, Ordering};
 
     #[derive(Debug, Clone, Copy, PartialEq, Eq)]
@@ -694,7 +695,7 @@ mod tests {
         }
     }
 
-    impl Groupped<MockHookEnum> for MockHookEnum {
+    impl Grouped<MockHookEnum> for MockHookEnum {
         fn member_id() -> MockHookEnum {
             MockHookEnum::A
         }