Fix for Async callback issue

Fixed the missing callback