Fix test mod