Fix unit test to work with pre-defined network