Simple Bug fix