Fixed broken test after #37 marge