Fixing bug in prev commit