Fix some missing test