Fixed bug in EM