Health inspectors have served an improvement notice on a distillery close to the site of an outbreak of Legionnaire's Disease in Edinburgh.