Fix callbacks tests