Re-add support for initialize callbacks