Update tests to reflect changes to examples