Add new phases to test entity driver