Fix up mistake from rename