Updating tests to cover a reported issue