fixed bug for the project