Changed test code