Test imports fix