[IMP]: fix small issue related to height