Fixed some test code