[TEST] - Fixing CI tests