Update tests for new syntax