Bug fixed in tests