Removing references to test_images