Add an `expect` here too?