Fixing test cases for adding distribution