Misc fixes..