Update test to use mtj