fix the bug with coding problem