Fixing correct/wrong bug