Fix a stuipid bug