Pequeña modificación para que funcione