Add a test for lifted loops