Fix json load in tests for python 3+