Add IF for rostest