Fixed error during build