Fix bug with setting kernel args in a few tests.