Fixed error in build