Correct module structure in tests