Fixes a bug
Bug fix