Fixing test case for QueueWrapper