Fixing broken build

Woops :smirk: