Fix for failing build