fix sample tests