Update tests to reflect new method airity