Made it work on azure