fixed the previous problem