Removing little mistake

removing little bug in the code