Re-added missing test