Fixes files to work in test