make the code correct