[SQUASH ME] Fix failing unittest