Fix bug with missing files