code coevrage fix, works locally