Updated to import correct org.junit.Assert class