Bug fixing

please prog @satnami fix the bug in code