Fix bug.
This is a really experimental commit. I am not a good programmer, but I have to fix these lol