adapt test to new args