Bug fix. 2