diff --git a/src/commands/pretty.ts b/src/commands/pretty.ts
index 35440fd6ff2..68525a2a99b 100644
--- a/src/commands/pretty.ts
+++ b/src/commands/pretty.ts
@@ -41,7 +41,7 @@ export default class Pretty extends Command {
 
     try {
       const fileFormat = retrieveFileFormat(this.specFile.text());
-      if (fileFormat === 'yaml') {
+      if (fileFormat === 'yaml' || fileFormat == 'yml') {
         const yamlDoc = yaml.parseDocument(content);
         formatted = yamlDoc.toString({
           lineWidth: 0,