[MAGEMEP-166] Fixed some bugs