Update tests for #3