Fixing small regression