Added to the test code