Update for missing test