Fix: Failing build

/cc @GeeH