Fixed small problem