fixed the bug!