Update tests to use new syntax