Maybe fix the failing assertion