Move wait variable to spec helper