Fix bug (but there might be many more like it)