fixed the first bug