Fix bug again. :(