fix a bit more

Change-Id: I8961fa9279a955f5570e58f29e23d412e5fd049c