Modify the code to make the code compil