another fix in the code