Fix more issues on feature unlocked screen