Minor bug fix

Please see my commit, it is an evident bug.