Added a new test and made it work.