Update tests for #61