Fix logic for choosing zk node