Restore missing test