fixng bug