Corrected import diagram