Fixed a small little bug