Reverted small fix