Fix to broken build