Fixed some broken test