Fix mistake in tests/ __init__ file