Fixed code

fixed