add back sample tests. oops.