Fix small errors in tests