add normalize back in