Bug fix in test scipt.