fixed bug for sure