diff --git a/lib/cli/commands-schema/aws-service.js b/lib/cli/commands-schema/aws-service.js
index ddc523239..c010ae49d 100644
--- a/lib/cli/commands-schema/aws-service.js
+++ b/lib/cli/commands-schema/aws-service.js
@@ -272,7 +272,7 @@ commands.set('studio', {
   lifecycleEvents: ['studio'],
 });
 
-commands.get('test', {
+commands.set('test', {
   usage: 'Run HTTP tests',
   options: {
     function: {
diff --git a/lib/cli/commands-schema/no-service.js b/lib/cli/commands-schema/no-service.js
index ad8d8b861..534c0778e 100644
--- a/lib/cli/commands-schema/no-service.js
+++ b/lib/cli/commands-schema/no-service.js
@@ -122,7 +122,7 @@ commands.set('dashboard', {
   lifecycleEvents: ['dashboard'],
 });
 
-commands.get('generate-event', {
+commands.set('generate-event', {
   usage: 'Generate event',
   lifecycleEvents: ['generate-event'],
   options: {