Fix pyflakes errors in make check

Fix 2 pyflakes errors in make check

Signed-off-by: Royce Lv <lvroyce@linux.vnet.ibm.com>