Updated test_init to follow new logic.