fixed little trouble