Update leaks unit test
Update leaks unit test