fixing a test to be more robust
added another test case