fixing first bug found