EntityTest.java update. New test :3