fix bugs
Small bug fixed