minor updates to test