Cleanup för easier testing