Fix for missing callback