correction to make the code compiling