Fixed the code