Update for testmod