Its better to replace the wrapper function