Small bug Fix