fixes broken test in #7