Fixed testing code.