Fixes module imports