small modification on wrapper