Use correct type annotations

1 files changed, 5 insertions, 6 deletions

diff --git a/patternutils/commands/patterncfg.py b/patternutils/commands/patterncfg.py
index 7c930da..01e53fc 100644
--- a/patternutils/commands/patterncfg.py
+++ b/patternutils/commands/patterncfg.py
@@ -1,6 +1,8 @@
 import os
 import os.path
 
+from typing import Optional
+
 import click
 
 from .. import config
@@ -31,11 +33,8 @@ def list() -> None:
 @click.option('-c', '--command', default=None, type=click.Choice(['pmatch', 'tmv', 'targs']),
               help='select the command from which to view the file')
 @click.argument('name')
-def add(
-    file_type: str,
-    command: str,
-    name: str,
-) -> None:
+def add(file_type: str, command: Optional[str], name: str) -> None:
+
     if command == 'pmatch':
         fileext = 'regex'
     else:
@@ -51,7 +50,7 @@ def add(
 @click.option('-c', '--command', default=None, type=click.Choice(['pmatch', 'tmv', 'targs']),
               help='select the command from which to view the file')
 @click.argument('name')
-def remove(command: str, name: str) -> None:
+def remove(command: Optional[str], name: str) -> None:
     files = config.find_templates(name, command)
     try:
         os.remove(files[0])