Improving tests over prompt