Minor bugfix after files changes