Fixed importing problem in testing modules