Fixed warning in tests