Small changes to keep it running on my azure account