Sorry, I didn't read the test close enough.  LGTM