Fix logic

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

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