Test code was out of date.  Updating