Adding fix for failing Python unit test.