Fixing mistake made in previous commit