diff --git a/test/selfhost-digitalocean-filesystem-test.js b/test/selfhost-digitalocean-filesystem-test.js
index f4c5297..d88ed6e 100644
--- a/test/selfhost-digitalocean-filesystem-test.js
+++ b/test/selfhost-digitalocean-filesystem-test.js
@@ -159,11 +159,13 @@ describe('Selfhost DigitalOcean with filesystem backend', function () {
         assert.strictEqual(out.status, 0);
 
         console.log('run cloudron-setup');
-        out = machine('ssh ' + params.concat(['"./cloudron-setup --provider digitalocean --prerelease --version ' + toVersion + ' --env staging"']).join(' '));//, { stdout: process.stdout, stderr: process.stderr });
+        out = machine('ssh ' + params.concat(['"./cloudron-setup --provider digitalocean --prerelease --version ' + toVersion + ' --env staging --skip-reboot"']).join(' '));
         console.log(out.stdout, out.stderr);
+        assert.strictEqual(out.status, 0);
 
-        // due to the reboot, we might get a non 0 status code
-        assert(out.status === 0 || out.stdout.indexOf('Rebooting this server now to let bootloader changes take effect.') !== -1);
+        console.log('reboot the server');
+        out = machine('ssh ' + params.concat(['systemctl reboot']).join(' '));
+        console.log(out.stdout, out.stderr);
     });
 
     it('can setup dns', function () {