fixed bug maybe I hope