Add same logic to only_with_rvm