Modify unittest