Remove this import and change the junit assert to scala assert?