updated test to make it more consistent