Remove test code in update