Fix forgotten files