#324 Fix test