Fix argument handling in test runner