[test] Fixing test