fix Testing functionality