fixed first bug!