fixed bug with testing