fixed a little problem