improve heka module