Fixed a lil bug