Small logic fix.