Fix test for six imports mapping