Restore original test code