Updated code to compile correctly.