fixed little problem