Fixing distribution tests i.e. waiting for some elements on distribution screen