Little update in tests