Fixed wrapper function