Just a few small fixes you may want for this wrapper.

Signed-off-by: ichbin <ichbin@ichbin.us>