Fixing another small issue