Fix collapse toggle