Memory Leak Fix

Memory leak fixed by adding missing callback