Update tests re: #138