Ridiculous hack to make test_distributions pass