Fix traceback

Signed-off-by: Chuck Short <chuck.short@canonical.com>