diff --git a/test/selfhost-ec2-create-test.js b/test/selfhost-ec2-create-test.js
index 443c4b3..3316ee8 100644
--- a/test/selfhost-ec2-create-test.js
+++ b/test/selfhost-ec2-create-test.js
@@ -129,7 +129,22 @@ describe('Selfhost EC2 Cloudron creation', function () {
     });
 
     it('can restore the box', function () {
-        var out = machine(util.format('restore ec2 --fqdn %s --type %s --region %s --ssh-key %s --access-key-id %s --secret-access-key %s --subnet %s --security-group %s --backup-key %s --backup-bucket %s --backup %s', CUSTOM_DOMAIN, EC2_TYPE, EC2_REGION, EC2_SSH_KEY, AWS_ACCESS_KEY, AWS_ACCESS_SECRET, EC2_SUBNET, EC2_SECURITY_GROUP, BACKUP_KEY, BACKUP_BUCKET, backupInfo.id));
+        var params = [
+            '--fqdn ' + CUSTOM_DOMAIN,
+            '--type ' + EC2_TYPE,
+            '--size ' + EC2_SIZE,
+            '--region ' + EC2_REGION,
+            '--ssh-key ' + EC2_SSH_KEY,
+            '--access-key-id ' + AWS_ACCESS_KEY,
+            '--secret-access-key ' + AWS_ACCESS_SECRET,
+            '--subnet ' + EC2_SUBNET,
+            '--security-group ' + EC2_SECURITY_GROUP,
+            '--backup-key ' + BACKUP_KEY,
+            '--backup-bucket ' + BACKUP_BUCKET,
+            '--backup ' + backupInfo.id,
+        ];
+
+        var out = machine('restore ec2 ' + params.join(' '));
 
         if (out.stdout.indexOf('You can now use your Cloudron at') === -1) {
             console.error(out.stdout, out.stderr);