Small bug fix. issueID #7