Fix tiny problem