Fixes for installtion screen