Quick fix to failing tests

cf: #44