1 files changed, 2 insertions, 14 deletions
diff --git a/runtime/probes/bench/bench_multi.c b/runtime/probes/bench/bench_multi.c
index 566e8520..eab75878 100644
--- a/runtime/probes/bench/bench_multi.c
+++ b/runtime/probes/bench/bench_multi.c
@@ -62,24 +62,12 @@ static struct kprobe kp[] = {
 
 #define NUM_KPROBES (sizeof(kp)/sizeof(struct kprobe))
 
-int init_module(void)
+int probe_start(void)
 {
-  int ret;
-
-  TRANSPORT_OPEN;
-  
-  ret = _stp_register_kprobes (kp, NUM_KPROBES);
-  return ret;
+  return _stp_register_kprobes (kp, NUM_KPROBES);
 }
 
 static void probe_exit (void)
 {
   _stp_unregister_kprobes (kp, NUM_KPROBES); 
 }
-
-void cleanup_module(void)
-{
-  _stp_transport_close();
-}
-
-MODULE_LICENSE("GPL");