Clean up ready for tests