[test] Fixing tests