fix a bug that I made