1 files changed, 2 insertions, 13 deletions
diff --git a/runtime/probes/tasklet/stp_tasklet.c b/runtime/probes/tasklet/stp_tasklet.c
index 46dacf03..2367d4d6 100644
--- a/runtime/probes/tasklet/stp_tasklet.c
+++ b/runtime/probes/tasklet/stp_tasklet.c
@@ -28,13 +28,9 @@ static struct jprobe stp_probes[] = {
 };
 #define MAX_STP_PROBES (sizeof(stp_probes)/sizeof(struct jprobe))
 
-int init_module(void)
+int probe_start(void)
 {
-  int ret;
-  
-  TRANSPORT_OPEN;
-  ret = _stp_register_jprobes (stp_probes, MAX_STP_PROBES);
-  return ret;
+  return _stp_register_jprobes (stp_probes, MAX_STP_PROBES);
 }
 
 static void probe_exit (void)
@@ -42,10 +38,3 @@ static void probe_exit (void)
   _stp_unregister_jprobes (stp_probes, MAX_STP_PROBES);
 }
 
-void cleanup_module(void)
-{
-  _stp_transport_close();
-}
-
-MODULE_LICENSE("GPL");
-