Fix the custom builder unit test

It was broken since ssh support was introduced.