Fix tiny bug