Fixes a minor bug