fix for di callback