Fix for heap tests.