Fix null unit test