Fixes tests for #119