Fixed testing code.
Fixed testing code.