- fixing junit test generation for new api