Fixed a small issue