Tweak of test now that it is working