fixing last small issue