CI failure fix