Add requirement for wait gem