Fixing test case after Ammar's comments.