Edited test code