Fixed the failing build