Return test instead of this if not in sample size

Should make it not return an error.