Fixes a little bug