Fixed code lose

Fixed code lose