FIx bug with my Code