Fixing callbacks tests