Small fix.

Small fix of previous update.