Update fix for working files