test: fixed distribution check