Fixing bug in previous commit