Make it run on mri too