Fixed the bug!