Minor Bug fix