Move code to medium tests.