Fix breaking build