Update tests for DigitalOcean