Added the test but I don't have time to fix the code right now.