Fix tests for shell argument