Only needs updated version of test code