Fix tests for changes in #74