Fixed broken tests. Relates to #73.