bug fixed! + mini bug to fix