Fix bug because I'm stupid