Update to use expect syntax