fix module importation issue in tests directory