Fix shell wrapper