Fixed slight bug