Correcting error in build.

I'm sorry I did not run the unit tests in my own machine.