Fixing ec unit tests