BUG fixed broken test case