[#58] what are we supposed to do now when the issue says improve tests?