Bug fix 3

(I shouldn't edit code in the github)