Fixed the code

fixed the code