[TEST]: Fixing ci tests