Fix unit test error
Merge pull request #582 from pulp/barnabycourt-956711-tests

Fix unit test error