Fixing bug with toggle collapse