Update tests for changes made in this branch