Fix correct files test