Fix small problem