Fixed broken test codes.