Fixing more coverage things