Fixed tests broken from CSS changes