diff --git a/src/renderer/components/InputField.tsx b/src/renderer/components/InputField.tsx
index eb754c8..6a9ce6a 100644
--- a/src/renderer/components/InputField.tsx
+++ b/src/renderer/components/InputField.tsx
@@ -1,5 +1,6 @@
 import { InputSchema } from 'src/shared/generated_models';
-import { match } from 'ts-pattern';
+import { match, NonExhaustiveError } from 'ts-pattern';
+import log from 'electron-log/renderer';
 import DirectoryField from './input_fields/DirectoryField';
 import FileField from './input_fields/FileField';
 import TextField from './input_fields/TextField';
@@ -17,6 +18,64 @@ type InputFieldProps = {
   disabled?: boolean;
 };
 
+function getFieldByInputSchema(
+  inputSchema: InputSchema,
+  value: any,
+  onChange: (value: any) => void,
+  disabled: boolean,
+) {
+  try {
+    return match(inputSchema)
+      .with({ inputType: 'text' }, () => (
+        <TextField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'textarea' }, () => (
+        <TextAreaField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'file' }, () => (
+        <FileField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'directory' }, () => (
+        <DirectoryField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'batchtext' }, () => (
+        <BatchTextField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'batchfile' }, () => (
+        <BatchFileField value={value} onChange={onChange} disabled={disabled} />
+      ))
+      .with({ inputType: 'batchdirectory' }, () => (
+        <BatchDirectoryField
+          value={value}
+          onChange={onChange}
+          disabled={disabled}
+        />
+      ))
+      .with(
+        {
+          inputType: {
+            inputType: 'newfile',
+          },
+        },
+        () => (
+          <NewFileField
+            inputSchema={inputSchema}
+            value={value}
+            onChange={onChange}
+            disabled={disabled}
+          />
+        ),
+      )
+      .exhaustive();
+  } catch (e) {
+    if (e instanceof NonExhaustiveError) {
+      log.error('Received an unexpected input schema.', inputSchema);
+      return <div>Unsupported input type.</div>;
+    }
+    throw e;
+  }
+}
+
 export default function InputField({
   inputSchema,
   value,
@@ -31,66 +90,7 @@ export default function InputField({
       {inputSchema.subtitle && (
         <p className="text-xs mt-1">{inputSchema.subtitle}</p>
       )}
-      {match(inputSchema)
-        .with({ inputType: 'text' }, () => (
-          <TextField value={value} onChange={onChange} disabled={disabled} />
-        ))
-        .with({ inputType: 'textarea' }, () => (
-          <TextAreaField
-            value={value}
-            onChange={onChange}
-            disabled={disabled}
-          />
-        ))
-        .with({ inputType: 'file' }, () => (
-          <FileField value={value} onChange={onChange} disabled={disabled} />
-        ))
-        .with({ inputType: 'directory' }, () => (
-          <DirectoryField
-            value={value}
-            onChange={onChange}
-            disabled={disabled}
-          />
-        ))
-        .with({ inputType: 'batchtext' }, () => (
-          <BatchTextField
-            value={value}
-            onChange={onChange}
-            disabled={disabled}
-          />
-        ))
-        .with({ inputType: 'batchfile' }, () => (
-          <BatchFileField
-            value={value}
-            onChange={onChange}
-            disabled={disabled}
-          />
-        ))
-        .with({ inputType: 'batchdirectory' }, () => (
-          <BatchDirectoryField
-            value={value}
-            onChange={onChange}
-            disabled={disabled}
-          />
-        ))
-        .with(
-          {
-            inputType: {
-              inputType: 'newfile',
-            },
-          },
-          () => (
-            <NewFileField
-              inputSchema={inputSchema}
-              value={value}
-              onChange={onChange}
-              disabled={disabled}
-            />
-          ),
-        )
-        .otherwise(() => (
-          <div>Unsupported input type.</div>
-        ))}
+      {getFieldByInputSchema(inputSchema, value, onChange, disabled)}
     </div>
   );
 }
diff --git a/src/renderer/components/ParameterField.tsx b/src/renderer/components/ParameterField.tsx
index 73c3611..e2a60e6 100644
--- a/src/renderer/components/ParameterField.tsx
+++ b/src/renderer/components/ParameterField.tsx
@@ -1,5 +1,7 @@
 import { ParameterSchema } from 'src/shared/generated_models';
-import { match } from 'ts-pattern';
+import { match, NonExhaustiveError } from 'ts-pattern';
+import log from 'electron-log/renderer';
+import { ReactNode } from 'react';
 import TextField from './parameter_fields/TextField';
 import IntField from './parameter_fields/IntField';
 import FloatField from './parameter_fields/FloatField';
@@ -7,93 +9,147 @@ import EnumField from './parameter_fields/EnumField';
 import RangedIntField from './parameter_fields/RangedIntField';
 import RangedFloatField from './parameter_fields/RangedFloatField';
 
-export default function ParameterField({
-  parameterSchema,
-  value,
-  onChange,
-  disabled = false,
+function LabelAndSubtitle({
+  children,
+  label,
+  subtitle,
 }: {
-  parameterSchema: ParameterSchema;
-  value: any;
-  onChange: (value: any) => void;
-  // eslint-disable-next-line
-  disabled?: boolean;
+  children: ReactNode;
+  label: string;
+  subtitle: string | null;
 }) {
-  const { value: paramValue } = parameterSchema;
   return (
     <div>
-      <h2 className="font-semibold text-sm xl:text-base mb-2">
-        {parameterSchema.label}
-      </h2>
-      {parameterSchema.subtitle && (
-        <p className="text-xs mt-1">{parameterSchema.subtitle}</p>
-      )}
-      {match(paramValue)
-        .with({ parameterType: 'text' }, () => {
-          return (
+      <h2 className="font-semibold text-sm xl:text-base mb-2">{label}</h2>
+      {subtitle && <p className="text-xs mt-1">{subtitle}</p>}
+      {children}
+    </div>
+  );
+}
+
+function getFieldByParameterSchema(
+  parameterSchema: ParameterSchema,
+  value: any,
+  onChange: (value: any) => void,
+  disabled: boolean,
+) {
+  try {
+    return match(parameterSchema.value)
+      .with({ parameterType: 'text' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <TextField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .with({ parameterType: 'int' }, () => {
-          return (
+          </LabelAndSubtitle>
+        );
+      })
+      .with({ parameterType: 'int' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <IntField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .with({ parameterType: 'float' }, () => {
-          return (
+          </LabelAndSubtitle>
+        );
+      })
+      .with({ parameterType: 'float' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <FloatField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .with({ parameterType: 'enum' }, () => {
-          return (
+          </LabelAndSubtitle>
+        );
+      })
+      .with({ parameterType: 'enum' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <EnumField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .with({ parameterType: 'ranged_int' }, () => {
-          return (
+          </LabelAndSubtitle>
+        );
+      })
+      .with({ parameterType: 'ranged_int' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <RangedIntField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .with({ parameterType: 'ranged_float' }, () => {
-          return (
+          </LabelAndSubtitle>
+        );
+      })
+      .with({ parameterType: 'ranged_float' }, () => {
+        return (
+          <LabelAndSubtitle
+            label={parameterSchema.label}
+            subtitle={parameterSchema.subtitle}
+          >
             <RangedFloatField
               parameterSchema={parameterSchema}
               value={value}
               onChange={onChange}
               disabled={disabled}
             />
-          );
-        })
-        .otherwise(() => {
-          return (
-            <div>Unsupported parameter type: {paramValue.parameterType}</div>
-          );
-        })}
-    </div>
-  );
+          </LabelAndSubtitle>
+        );
+      })
+      .exhaustive();
+  } catch (e) {
+    if (e instanceof NonExhaustiveError) {
+      log.error('Received an unexpected parameter schema.', parameterSchema);
+      return (
+        <div>
+          Unsupported parameter type: {parameterSchema.value.parameterType}
+        </div>
+      );
+    }
+    throw e;
+  }
+}
+
+export default function ParameterField({
+  parameterSchema,
+  value,
+  onChange,
+  disabled = false,
+}: {
+  parameterSchema: ParameterSchema;
+  value: any;
+  onChange: (value: any) => void;
+  // eslint-disable-next-line
+  disabled?: boolean;
+}) {
+  return getFieldByParameterSchema(parameterSchema, value, onChange, disabled);
 }
diff --git a/src/renderer/jobs/PreviewResponseBody.tsx b/src/renderer/jobs/PreviewResponseBody.tsx
index cc9a351..5ea7ed7 100644
--- a/src/renderer/jobs/PreviewResponseBody.tsx
+++ b/src/renderer/jobs/PreviewResponseBody.tsx
@@ -1,6 +1,6 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 import { ResponseBody } from 'src/shared/generated_models';
-import { match } from 'ts-pattern';
+import { match, NonExhaustiveError } from 'ts-pattern';
 import MarkdownView from '../components/response_body/text_views/MarkdownView';
 import DirectoryView from '../components/response_body/directory_views/DirectoryView';
 import BatchDirectoryView from '../components/response_body/directory_views/BatchDirectoryView';
@@ -14,29 +14,34 @@ export default function PreviewResponseBody({
 }: {
   response: ResponseBody;
 }) {
-  return match(response)
-    .with({ output_type: 'file' }, (fileResponse) => {
-      return <FileView response={fileResponse} />;
-    })
-    .with({ output_type: 'directory' }, (directoryResponse) => {
-      return <DirectoryView response={directoryResponse} />;
-    })
-    .with({ output_type: 'markdown' }, (markdownResponse) => {
-      return <MarkdownView response={markdownResponse} />;
-    })
-    .with({ output_type: 'text' }, (textResponse) => {
-      return <TextView response={textResponse} />;
-    })
-    .with({ output_type: 'batchfile' }, (batchFileResponse) => {
-      return <BatchFileView response={batchFileResponse} />;
-    })
-    .with({ output_type: 'batchtext' }, (batchTextResponse) => {
-      return <BatchTextView response={batchTextResponse} />;
-    })
-    .with({ output_type: 'batchdirectory' }, (batchDirectoryResponse) => {
-      return <BatchDirectoryView response={batchDirectoryResponse} />;
-    })
-    .otherwise(() => {
+  try {
+    return match(response)
+      .with({ output_type: 'file' }, (fileResponse) => {
+        return <FileView response={fileResponse} />;
+      })
+      .with({ output_type: 'directory' }, (directoryResponse) => {
+        return <DirectoryView response={directoryResponse} />;
+      })
+      .with({ output_type: 'markdown' }, (markdownResponse) => {
+        return <MarkdownView response={markdownResponse} />;
+      })
+      .with({ output_type: 'text' }, (textResponse) => {
+        return <TextView response={textResponse} />;
+      })
+      .with({ output_type: 'batchfile' }, (batchFileResponse) => {
+        return <BatchFileView response={batchFileResponse} />;
+      })
+      .with({ output_type: 'batchtext' }, (batchTextResponse) => {
+        return <BatchTextView response={batchTextResponse} />;
+      })
+      .with({ output_type: 'batchdirectory' }, (batchDirectoryResponse) => {
+        return <BatchDirectoryView response={batchDirectoryResponse} />;
+      })
+      .exhaustive();
+  } catch (e) {
+    if (e instanceof NonExhaustiveError) {
       return <div>Unknown Response</div>;
-    });
+    }
+    throw e;
+  }
 }
diff --git a/src/renderer/models/ModelRunTask.tsx b/src/renderer/models/ModelRunTask.tsx
index efdc3e2..6633c46 100644
--- a/src/renderer/models/ModelRunTask.tsx
+++ b/src/renderer/models/ModelRunTask.tsx
@@ -39,6 +39,14 @@ export default function ModelRunTask() {
     return <div>Invalid Model UID or Task ID.</div>;
   }
 
+  if (taskSchemaIsLoading) {
+    return <LoadingScreen />;
+  }
+
+  if (taskSchemaError) {
+    return <div>Error loading task schema</div>;
+  }
+
   if (!thisApiRoute || !taskSchema) {
     return (
       <div
@@ -60,12 +68,6 @@ export default function ModelRunTask() {
       </div>
     );
   }
-  if (taskSchemaIsLoading) {
-    return <LoadingScreen />;
-  }
-  if (taskSchemaError) {
-    return <div>Error loading task schema</div>;
-  }
 
   const onSubmit = async (data: any) => {
     const runJobArgs: RunJobArgs = {
@@ -168,11 +170,6 @@ export default function ModelRunTask() {
             ))}
           </div>
         </div>
-        // <div className="p-2 rounded-md bg-yellow-200">
-        //   <h2 className="text-xl font-bold mb-4">Not Runnable</h2>
-        //   <ul className="list-disc list-inside">
-        //   </ul>
-        // </div>
       )}
       <Button
         type="submit"
diff --git a/src/shared/dummy_data/set_dummy_mode.ts b/src/shared/dummy_data/set_dummy_mode.ts
index 48bc3a1..f34696a 100644
--- a/src/shared/dummy_data/set_dummy_mode.ts
+++ b/src/shared/dummy_data/set_dummy_mode.ts
@@ -1,3 +1,3 @@
-const isDummyMode = true;
+const isDummyMode = false;
 
 export default isDummyMode;