fix bug with Tests
now fix bug with Tests