Fix module path in tests