Fix bug + add testdata