fix final bug