Make it work with any ending