Fix tiny issue