Fix the remaining failing test and add the implementation to make it pass.