registration: remove unnecessary .flatten() call