Fixed error breaking build