Temporary fix for failing test