Fix small fail