Fixed the same bug in tests