Import correct module