Bug fixed

This is fix for a bug.