Fixes/tweak