Fix error in recent update