Fix last commit.
Another fix for sme bug.
It's late at night.