Add the missing test