Fixed breaking build (hopefully)