Tiny logic fix