adapt tests to NormalizingLayer changes