Fix for broken build