Bug fix

Fixed a little bug caused by a github pull.