Add rex-mode tests