small fix
adding wrapper
fix