Fixing testing code that was accidentally committed.