Update spec to use expect syntax