Fix failed test

Due to #1092