litlle bug fixed