fixed isue
amending code