fixing those tests