Bug fix - silly edit