Adapted test class for maven