fixed a small issue..