- quick fix for update logic