Old implementation was showing import error, new implementation instead hides it behind separated call and if-else check