Fix controller update