fix: safe imports for proof callbacks