Adding fixes for failures due to modifications of automation framework

Adding fixes for failures due to modifications of automation framework