Fix python3 unit test

Change-Id: I052c1474e5993ff47f3b0c325868001bebe4ff3c