Fixing missing callback