Fix for multi-line tests