CS fixes

Change-Id: I3dedaa59f1cfb244e14e2cbffdbde06be32bba3a