fix errors in tests