Fixed broken test from adding in the JsonFormatter

Fixed code so it worked for test relying on JsonFormatter.