Litlle bug fixed