Fix up a bug pointed out by pylint.

Signed-off-by: Chris Lalancette <clalancette@gmail.com>