Fixed code for proper compilation.