Fix tests. Close #19.