Fix missing files from last commit