Fixed an issue with .with() in tests