[Core] Fix imports