Fixing import error in mp2 module