Update tests for #158