fixed little bug