add back the end hack to the tests, travis getting hung up for some reason