Updated test to be more accurate