small problem fix