Fix failed test